'''
Mission 4 -- Display
Final code with addition of results() function
'''
from codex import *
from time import sleep
import random

button_list = [ ["Hold BTN-A", BTN_A], ["Hold BTN-B", BTN_B],
                ["Hold UP", BTN_U], ["Hold DOWN", BTN_D],
                ["Hold RIGHT", BTN_R], ["Hold LEFT", BTN_L] ]

def play_game(level):
    correct = 0
    if level == 1:
        delay = 1.5
        tries = 5
    elif level == 2:
        delay = 1.0
        tries = 10
    else:
        delay = 0.5
        tries = 15
    for count in range(tries):
        item = random.choice(button_list)
        message = item[0]
        button = item[1]
        display.show(message)
        sleep(delay)
        pressed = buttons.is_pressed(button)
        if pressed:
            pixels.fill(GREEN)
            correct = correct + 1
        else:
            pixels.fill(RED)
        # clear button presses
        buttons.was_pressed()
        time.sleep(1)
    results(correct, tries)

def results(correct, tries):
    display.clear()
    percent = int(correct / tries * 100)
    display.print("Results:", scale=3)
    display.print("Number correct:", color=GREEN)
    display.print(str(correct), scale=5, color=GREEN)
    display.print("Percentage:", color=CYAN)
    display.print(str(percent), scale=5, color=CYAN)
    display.print()
    if percent == 100:
        display.print("Perfect score", color=PINK)
        pix_color = PINK
    elif percent > 60:
        display.print("Good job", color=BLUE)
        pix_color = BLUE
    elif percent > 40:
        display.print("Getting closer", color=PURPLE)
        pix_color = PURPLE
    else:
        display.print("Not good", color=RED)
        pix_color = RED
    for i in range(5):
        pixels.fill(pix_color)
        sleep(0.2)
        pixels.off()
        sleep(0.1)

# -- Main Program
while True:
    if buttons.was_pressed(BTN_L):
        play_game(1)
    if buttons.was_pressed(BTN_U):
        play_game(2)
    if buttons.was_pressed(BTN_R):
        play_game(3)

    if buttons.was_pressed(BTN_D):
        pixels.off()
        break